narrow network
A Neural Scaling Law from Lottery Ticket Ensembling
Neural scaling laws (NSL) refer to the phenomenon where model performance improves with scale. Sharma & Kaplan analyzed NSL using approximation theory and predict that MSE losses decay as $N^{-\alpha}$, $\alpha=4/d$, where $N$ is the number of model parameters, and $d$ is the intrinsic input dimension. Although their theory works well for some cases (e.g., ReLU networks), we surprisingly find that a simple 1D problem $y=x^2$ manifests a different scaling law ($\alpha=1$) from their predictions ($\alpha=4$). We opened the neural networks and found that the new scaling law originates from lottery ticket ensembling: a wider network on average has more "lottery tickets", which are ensembled to reduce the variance of outputs. We support the ensembling mechanism by mechanistically interpreting single neural networks, as well as studying them statistically. We attribute the $N^{-1}$ scaling law to the "central limit theorem" of lottery tickets. Finally, we discuss its potential implications for large language models and statistical physics-type theories of learning.
When Expressivity Meets Trainability: Fewer than $n$ Neurons Can Work
Zhang, Jiawei, Zhang, Yushun, Hong, Mingyi, Sun, Ruoyu, Luo, Zhi-Quan
Modern neural networks are often quite wide, causing large memory and computation costs. It is thus of great interest to train a narrower network. However, training narrow neural nets remains a challenging task. We ask two theoretical questions: Can narrow networks have as strong expressivity as wide ones? If so, does the loss function exhibit a benign optimization landscape? In this work, we provide partially affirmative answers to both questions for 1-hidden-layer networks with fewer than $n$ (sample size) neurons when the activation is smooth. First, we prove that as long as the width $m \geq 2n/d$ (where $d$ is the input dimension), its expressivity is strong, i.e., there exists at least one global minimizer with zero training loss. Second, we identify a nice local region with no local-min or saddle points. Nevertheless, it is not clear whether gradient descent can stay in this nice region. Third, we consider a constrained optimization formulation where the feasible region is the nice local region, and prove that every KKT point is a nearly global minimizer. It is expected that projected gradient methods converge to KKT points under mild technical conditions, but we leave the rigorous convergence analysis to future work. Thorough numerical results show that projected gradient methods on this constrained formulation significantly outperform SGD for training narrow neural nets.
Width is Less Important than Depth in ReLU Neural Networks
Vardi, Gal, Yehudai, Gilad, Shamir, Ohad
We solve an open question from Lu et al. (2017), by showing that any target network with inputs in $\mathbb{R}^d$ can be approximated by a width $O(d)$ network (independent of the target network's architecture), whose number of parameters is essentially larger only by a linear factor. In light of previous depth separation theorems, which imply that a similar result cannot hold when the roles of width and depth are interchanged, it follows that depth plays a more significant role than width in the expressive power of neural networks. We extend our results to constructing networks with bounded weights, and to constructing networks with width at most $d+2$, which is close to the minimal possible width due to previous lower bounds. Both of these constructions cause an extra polynomial factor in the number of parameters over the target network. We also show an exact representation of wide and shallow networks using deep and narrow networks which, in certain cases, does not increase the number of parameters over the target network.
Infinite Neural Networks!
TL;DR -- Recent research shows that'wide' neural networks change very little when they are trained, while'narrow' networks change the weights of their synapses dramatically. This is a consequence of the fact that those wide-nets tend to all turn into the same network, statistically. Because all initializations are the same, this reduces the'distance' a network must travel to minimize loss. So, less change occurs during training. Though this makes them train faster and with greater expressiveness, it's actually counter-productive to generalization and following symbolic constraints and analogies, as well as for concatenating sub-tasks into goals. We'll need a Mixture of Experts for those problems.
[R] Neural networks vs The Game of Life
I think I see what you're saying, but... The way the paper reads, the problem tackled is: "Given sample trajectories from Game of Life but no access to / knowledge of the source code, create a model that perfectly predicts the game's dynamic." However, by using n to set the problem difficulty (by using a terminal loss instead of a trajectory loss), the actual problem being tackled is: "Given sample state pairs {y[0], y[n]} spaced n steps apart... create a model that perfectly predicts the game's dynamic." The latter is clearly a much more difficult problem. I can agree that a narrow network may have difficulties with it (perhaps related to the lottery hypothesis).
The Expressive Power of Neural Networks: A View from the Width
Lu, Zhou, Pu, Hongming, Wang, Feicheng, Hu, Zhiqiang, Wang, Liwei
The expressive power of neural networks is important for understanding deep learning. Most existing works consider this problem from the view of the depth of a network. In this paper, we study how width affects the expressiveness of neural networks. Classical results state that depth-bounded (e.g. We show a universal approximation theorem for width-bounded ReLU networks: width-(n 4) ReLU networks, where n is the input dimension, are universal approximators.